import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;

/**
 * 服务端转发消息给所有客户端
 * 当一个客户端发送一个消息后,服务端收到后如何转发给所有客户端
 * 解决:
 *  内部类可以访问外部类的成员,因此在Server类上定义一个集合allOut可以被所有内部类ClientHandler实例访问.
 *  从而将这些ClientHandler实例之间向互访的数据存放在这个集合中达到共享数据的目的.对此只需要将所有ClientHandler
 *  中的输出流都存入到集合allOut中就可以达到互访输出流转发消息的目的了
 */
public class Server {
    private ServerSocket serverSocket;

    /**
     * 初始化服务端
     */
    public Server() {
        try {
            System.out.println("正在启动服务端...");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 启动服务端的方法
     */
    public void start() {

        /**
         * ServerSocket提供了接受客户端连接的方法
         * Socket accept()
         * 该方法是一个阻塞方法,调用后方法"卡住",此时开始等待客户端连接,知道一个客户端连接,此时
         * 方法会返回一个Socket实例.通过这个Socket就可以与客户端进行交互了
         */
        try {
            Socket socket = null;
            while (true) {
                System.out.println("等待客户端连接...");
                socket = serverSocket.accept();
                System.out.println("一个客户端连接了!");
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }
}

class ClientHandler implements Runnable {
    //存放所有客户端输出流,用于广播消息
    private List<PrintWriter> allOut = new ArrayList();
    private Socket socket;
    private String host;    //记录客户端的ip地址

    public ClientHandler(Socket socket) {
        this.socket = socket;
        host = socket.getInetAddress().getHostAddress();
    }

    @Override
    public void run() {
        try {
            InputStream in = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(in);
            BufferedReader br = new BufferedReader(isr);

            OutputStream out = socket.getOutputStream();
            OutputStreamWriter osw = new OutputStreamWriter(out);
            BufferedWriter bw = new BufferedWriter(osw);
            PrintWriter pw = new PrintWriter(bw,true);
            allOut.add(pw);

            String s = null;
            while ((s = br.readLine()) != null) {
                System.out.println(host+"说:" + s);
//                pw.println(s);

                //将消息回复给所有客户端
                for (PrintWriter o : allOut){
                    o.println(host+"说:"+s);
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
